Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency
打破气泡:有界权重不一致的异步流水线并行训练
🧭 分布式训练 · 流水线并行优化
Technion - Israel Institute of Technology · Ben-Gurion University of the Negev
通过局部梯度积累作为版本控制,零气泡异步流水线以有界权重漂移换取训练加速
📌 工作参考价值(4 点)
- 梯度累积被重新用作一种轻量级版本控制机制:通过增大累积步数减缓参数版本演化,使异步流水线中微批次跨越的优化器更新次数有界,从而控制权重不一致,这个思路可直接用于训练推荐大模型或 Agent 的 LLM 骨干网,无需额外校正模块。
- PACI 无需权重暂存、预测或全局同步,内存峰值与同步 1F1B 持平,对 GPU 显存敏感的电商推荐模型训练(如多模态大模型)很有吸引力,可以在不增加显存成本的前提下消除流水线气泡。
- 实现零气泡流水线调度:所有设备持续利用,在 GPT 预训练中获得 1.69× 的时间到精度加速。对于需要快速迭代训练的大规模 Agent 或生成式推荐模型,可考虑采用类似有界异步流水线策略来压缩实验周期。
- 该方法揭示一个原则:前向/后向的权重不一致不必完全消除,显式控制其边界即可安全地换取效率。在设计其他分布式训练策略(如异步数据并行或混合并行)时,可借鉴这一思路,通过参数版本漂移的定量控制来简化系统设计。